scikit_learn
pandas
datasets
accelerate
transformers==4.39.0
pillow<10.0.0
numpy<2.0
